package com.zk.algorithm.backtracking;

import com.zk.algorithm.annotation.CrackingTheCodingInterview6;
import com.zk.algorithm.annotation.LeetCode;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @author zk
 */
@LeetCode
@CrackingTheCodingInterview6("8.8")
public class Permutations2 {

    // 能够包含重复
    // 1,1,2
    //
    // [1]
    //   [1,1]
    //     [1,1,1]
    //     [1,1,1]
    //     [1,1,2]
    //   [1,1]
    //     [1,1,1]
    //     [1,1,1]
    //     [1,1,2]
    //   [1,2]
    //     [1,2,1]
    //     [1,2,1]
    //     [1,2,2]
    // [1]
    //   [1,1]
    //     [1,1,1]
    //     [1,1,2]
    //     [1,1,3]
    //   [1,1]
    //     [1,1,1]
    //     [1,1,2]
    //     [1,1,3]
    //   [1,2]
    //     [1,2,1]
    //     [1,2,1]
    //     [1,2,2]
    // [2]
    //   [2,1]
    //     [2,1,1]
    //     [2,1,1]
    //     [2,1,2]
    //   [2,1]
    //     [2,1,1]
    //     [2,1,1]
    //     [2,1,2]
    //   [2,2]
    //     [2,2,1]
    //     [2,2,1]
    //     [2,2,2]
    public List<List<Integer>> permuteUnique(int[] nums) {
        List<List<Integer>> list = new ArrayList<>();
        Arrays.sort(nums);
        backtrack(list, new ArrayList<>(), nums, new boolean[nums.length]);
        return list;
    }

    private void backtrack(List<List<Integer>> list, List<Integer> tempList, int[] nums, boolean[] used) {
        if (tempList.size() == nums.length) {
            list.add(new ArrayList<>(tempList));
        } else {
            for (int i = 0; i < nums.length; i++) {
                if (used[i] || i > 0 && nums[i] == nums[i - 1] && !used[i - 1])
                    continue;

                // ========================
                // 末尾添加
                // ========================
                used[i] = true;
                tempList.add(nums[i]);
                // ========================
                // 回溯
                // ========================
                backtrack(list, tempList, nums, used);
                // ========================
                // 末尾去除
                // ========================
                used[i] = false;
                tempList.remove(tempList.size() - 1);
            }
        }
    }

}
